#pragma once
#include <iostream>
#include "nocopy.hpp"
#include <string>
#include <string.h>
#include <strings.h>
#include <errno.h>
#include<functional>
#include "comm.hpp"
#include "Log.hpp"
#include "inetAddr.hpp"
// 网络编程常用头文件
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

const static uint16_t default_port = 8888;
const static int default_fd = -1;

using func_t = std::function<std::string(std::string)>;

class UdpServer : public NoCopy 
{
    // 继承了NoCopy类;
public:
    UdpServer(func_t Func, uint16_t port = default_port)
        : _port(port), _sock_fd(default_fd), _Func(Func)
    {
    }
    ~UdpServer()
    {
    }
    void Init()
    {
        _sock_fd = socket(AF_INET, SOCK_DGRAM, 0); // 网络通信，UDP套接字
        if (_sock_fd < 0)
        {
            // 出错信息，打日志

            lg.LogMessage(Fatal, "socket error, %d : %s\n", errno, strerror(errno));
            exit(Socket_Err);
        }
        lg.LogMessage(Info, "success to init the socket, socket_fd: %d", _sock_fd);

        // 填充sockaddr结构体对象
        struct sockaddr_in local;
        bzero(&local, sizeof(local));       // 初始化结构体

        local.sin_family = AF_INET;             // 绑定网络通信
        local.sin_port = htons(_port);          // 绑定端口
        local.sin_addr.s_addr = INADDR_ANY;     // 绑定ip，将点分十进制转化为地址

        // 绑定指定网络信息和指定的文件系统
        int n = ::bind(_sock_fd, (struct sockaddr *)&local, sizeof(local));
        if (n != 0)
        {
            lg.LogMessage(Fatal, "Bind error, %d : %s", errno, strerror(errno));
            exit(Bind_Err);
        }

        lg.LogMessage(Info, "Bind success"); // 成功绑定
    }
    void Start()
    {
        // buffer receive
        char buff_r[1024];
        // 服务器启动时，不退出
        while (1)
        {
            sockaddr_in peer;
            socklen_t len = sizeof(peer);
            ssize_t n = recvfrom(_sock_fd, buff_r, sizeof(buff_r)-1, 0, (struct sockaddr *)&peer, &len);
            if (n > 0)
            {
                buff_r[n] = 0;
                InetAddr in_addr(peer);
                // std::cout << "[" << in_addr.GetAddr() << "]" << std::endl;
                std::cout << "[" << in_addr.GetAddr() << "]" << "client say: " << buff_r << std::endl;
                // 服务器向客户端发送信息
                // buffer send
                std::string buff_s = _Func(buff_r);
                sendto(_sock_fd, buff_s.c_str(), buff_s.size(), 0, (struct sockaddr *)&peer, len);
            }
        }
    }

private:
    // std::string _ip; 服务器不需要ip地址
    uint16_t _port;
    int _sock_fd;
    func_t _Func;
};